package com.jsj.subway.ai.agent.rag.models;

import lombok.Data;

@Data
public class KnowledgeBase {

    @Data
    public static class CreateReq{
        String kb_name;
        public CreateReq(String kb_name){
            this.kb_name = kb_name;
        }
    }

    @Data
    public static class UpdateReq{
        String kb_name;
        String kb_info;
    }

    @Data
    public static class DeleteReq{
        String kb_name;
        public DeleteReq(String kb_name){
            this.kb_name = kb_name;
        }
    }

    @Data
    public static class DeleteDocReq{
        String kb_name;
        Long id;
    }

    @Data
    public static class QueryReq{
        String kb_name;
        String query;
        Integer top_n;
        Float threshold;
        public QueryReq(String kb_name, String query) {
            this.kb_name = kb_name;
            this.query = query;
            this.top_n = 3;
        }
    }
}
